package com.cn.parks.servlet;

import com.cn.parks.entity.User;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import java.io.IOException;

@WebServlet("/userInfo")
public class UserInfoServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) 
            throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");
        HttpSession session = request.getSession(false);
        
        if (session != null && session.getAttribute("user") != null) {
            User user = (User) session.getAttribute("user");
            String userInfo = String.format(
                "{\"success\":true,\"data\":{\"userName\":\"%s\",\"roleId\":\"%s\"}}",
                user.getUserName(),
                user.getRoleId()
            );
            response.getWriter().write(userInfo);
        } else {
            response.getWriter().write("{\"success\":false,\"message\":\"未登录\"}");
        }
    }
} 